package cc.shacocloud.luanniao.core.service.ai.doubao

import com.volcengine.ark.runtime.model.Usage

/**
 * @author 思追(shaco)
 */
class DoubaoUsage(val usage: Usage) : org.springframework.ai.chat.metadata.Usage {
    override fun getPromptTokens(): Long {
        return usage.promptTokens
    }

    override fun getGenerationTokens(): Long {
        return usage.completionTokens
    }

    override fun getTotalTokens(): Long {
        return usage.totalTokens
    }

}